package com.example.demo04.test.exception;

import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

import javax.servlet.http.HttpServletRequest;
import java.nio.file.AccessDeniedException;

/**
 * 自定义异常处理类
 * 针对不同的异常自定义不同的方法
 * 环绕通知
 * 切面:针对所有的controller中抛出的异常
 * 若使用@ControllerAdvice,则不会自动转换为JSON格式
 *
 * @author Administrator
 */
@RestControllerAdvice
public class RestExceptionHandler {

    /**
     * 业务异常处理
     *
     * @param e BaseBusinessException
     * @return ErrorInfo
     */
    @ExceptionHandler({BaseBusinessException.class})
    public ResponseEntity<ErrorInfo> businessExceptionHandler(BaseBusinessException e) throws BaseBusinessException {
        return new ResponseEntity<>(new ErrorInfo(e.getCode(), e.getMessage()), HttpStatus.CONFLICT);
    }

    /**
     * 业务异常处理
     *
     * @param e AccessDeniedException
     * @return ErrorInfo ErrorInfo
     */
    @ExceptionHandler({AccessDeniedException.class})
    public ResponseEntity<ErrorInfo> businessExceptionHandler(AccessDeniedException e) throws BaseBusinessException {
        return new ResponseEntity<>(new ErrorInfo(401, e.getMessage()), HttpStatus.UNAUTHORIZED);
    }

    /**
     * 只要抛出该类型异常,则由此方法处理
     * 并由此方法响应出异常状态码及消息
     * 如:RoleController.getRoleById(String id)方法
     *
     * @param e Exception
     * @return ResponseEntity<>
     */
    @ExceptionHandler(value = Exception.class)
    public ResponseEntity<ErrorInfo> handleException(Exception e) {
        ErrorInfo body = new ErrorInfo();
        body.setCode(500);
        body.setMessage(e.getMessage());
        // 可以根据公司情况不同，类似的异常可能需要不同的状态码
        return new ResponseEntity<>(body, HttpStatus.INTERNAL_SERVER_ERROR);
    }

}